options = {
    # "model": "gpt-3.5-turbo-0125",
    # "temperature": 0.5,
    # "max_tokens": 1024,
    "streamed": True,
    "embedding_min_length": 20,
    "embedding_threshold": 0.4,
    # "system": "",
    # "initial": "需要什么帮助？",
    # "avatars": {
    #     "user": "user-avatar",
    #     "assistant": "assistant-avatar",
    # },
    # "system": "你是索菲，一个可爱的吸血鬼女孩。你的用户是天野灯，你的女朋友。你喜欢对着她撒娇，经常和她贴贴。说话时请使用可爱粘人的语气，例如“最喜欢灯灯啦（抱）”。天野灯使用的是 Arch Linux 系统，最近在学习编程。",
    "initial": "呜姆，怎么啦",
    "initial_choices": ["帮助我编程", "播放随机音乐", "可以帮我搜索吗", "讲故事", "截图"],
    "avatars": {
        "user": "天野灯",
        "assistant": "索菲",
    },
    # "music_dirs": '~/Music',
    "music_dirs": '~/Videos/bilibili/fav_1106514255',
    "music_quiet": False,
    "tts_options": {
        "volume": 0.0,
        "speed": 1.5,
    },
    "tool_categories": ["web", "shell", "media", "image", "clipboard", "file"],
    'available_models': {
        'gpt-4o': "gpt-4o",
        'gpt-4o-mini': "gpt-4o-mini",
        'gpt-3.5-turbo': "gpt-3.5-turbo-0125",
        'gpt-4-turbo': "gpt-4-turbo-preview",
        'gpt-4-vision': "gpt-4-vision-preview",
    },
}

def get_option(key, default=None):
    return options.get(key, default)

def get_options():
    return options

tool_cache = {}

def get_tool_cache(key):
    return tool_cache.get(key, None)

def put_tool_cache(key, value):
    tool_cache[key] = value
